import numpy as np
data = np.sin(np.arange(20)).reshape(5, 4)
# >>> data
# array([[ 0.        ,  0.84147098,  0.90929743,  0.14112001],
# [-0.7568025 , -0.95892427, -0.2794155 ,  0.6569866 ],
# [ 0.98935825,  0.41211849, -0.54402111, -0.99999021],
# [-0.53657292,  0.42016704,  0.99060736,  0.65028784],
# [-0.28790332, -0.96139749, -0.75098725,  0.14987721]])


# 求出每列最大值元素所在的行下标。
def action():
    return None


def test_action():
    assert (action() == [2, 0, 3, 1]).all()
